ubuntu 安装cuda 9.1 pytorch 0.3.0

毕业再没用配过机器学习的环境了，既亲切又陌生，久违了。

系统 mint18  x64

1安装cuda

  按官网提示 选的9.1版  https://developer.nvidia.com/cuda-toolkit

2安装Anaconda3

https://www.cnblogs.com/xuanmanstein/p/13892149.html

 

3Pytorch

直接在官网按按钮选版本http://pytorch.org/

如果没切换清华源，可能会奇慢无比。

 

结果还是极其慢，我倒。

只好从源码编译了

参考这个http://blog.csdn.net/draco_mystack/article/details/71191924 和pytorch的github https://github.com/pytorch/pytorch#from-source

CMAKE_PREFIX_PATH="$(dirname $(which conda))/../"

conda install numpy pyyaml mkl setuptools cmake cffi  
（还好，这几个清华的镜像站里都有）

conda install -c soumith magma-cuda90
（我的cuda下载nv官网的9.1 参考pytorch官网生成的安装命令行，所以弄了个90.原文是80）

虽然这一步还是有点慢，但是已经可以忍受了

 还是特别慢。

 

最后参考https://github.com/dnouri/skorch

直接从anaconda里起控制台，pip安装了

pip install http://download.pytorch.org/whl/cu90/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl